package com.xie.sort.selectsort;

import java.util.Arrays;

/**
 * 堆排序
 * <p>
 * 堆排序是借助堆来实现的选择排序，思想同简单的选择排序，以下以大顶堆为例。
 * 注意：如果想升序排序就使用大顶堆，反之使用小顶堆。
 * 原因是堆顶元素需要交换到序列尾部。
 * <p>
 * 　　首先，实现堆排序需要解决两个问题：
 * <p>
 * 　　1. 如何由一个无序序列键成一个堆？
 * <p>
 * 　　2. 如何在输出堆顶元素之后，调整剩余元素成为一个新的堆？
 * <p>
 *    第一个问题，可以直接使用线性数组来表示一个堆，
 *    由初始的无序序列建成一个堆就需要自底向上从第一个非叶元素开始挨个调整成一个堆。
 * <p>
 *    第二个问题，怎么调整成堆？
 *    首先是将堆顶元素和最后一个元素交换。
 *    然后比较当前堆顶元素的左右孩子节点，
 *    因为除了当前的堆顶元素，左右孩子堆均满足条件，
 *    这时需要选择当前堆顶元素与左右孩子节点的较大者（大顶堆）交换，
 *    直至叶子节点。我们称这个自堆顶自叶子的调整成为筛选。
 * <p>
 *    从一个无序序列建堆的过程就是一个反复筛选的过程。
 *    若将此序列看成是一个完全二叉树，
 *    则最后一个非终端节点是n/2取底个元素，
 *    由此筛选即可。
 */
public class HeapSort {

    public static void main(String[] args) {
        int[] arr = {49, 38, 65, 97, 23, 22, 76, 1, 5, 8, 2, 0, -1, 22};
        System.out.println("排序前:\n" + Arrays.toString(arr));
        // 堆排序
        heapSort(arr);
        System.out.println("排序后:\n" + Arrays.toString(arr));
    }

    /**
     * 堆筛选，除了start之外，start~end均满足大顶堆的定义。
     * 调整之后start~end称为一个大顶堆。
     *
     * @param arr   待调整数组
     * @param start 起始指针
     * @param end   结束指针
     */
    public static void heapAdjust(int[] arr, int start, int end) {
        int temp = arr[start];

        for (int i = 2 * start + 1; i <= end; i *= 2) {
            //左右孩子的节点分别为2*i+1,2*i+2

            //选择出左右孩子较小的下标
            if (i < end && arr[i] < arr[i + 1]) {
                i++;
            }
            if (temp >= arr[i]) {
                //已经为大顶堆，=保持稳定性。
                break;
            }
            //将子节点上移
            arr[start] = arr[i];
            //下一轮筛选
            start = i;
        }
        //插入正确的位置
        arr[start] = temp;
    }


    public static void heapSort(int[] arr) {
        if (arr == null || arr.length == 0) {
            return;
        }

        //建立大顶堆
        for (int i = arr.length / 2; i >= 0; i--) {
            heapAdjust(arr, i, arr.length - 1);
        }

        for (int i = arr.length - 1; i >= 0; i--) {
            swap(arr, 0, i);
            heapAdjust(arr, 0, i - 1);
        }

    }

    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

}
